#!/bin/sh

case $BUTTON in
	3) notify-send "🌐 网络流量模块" "\- 查看实时网络流量
- 🔼: 下行流量
- 🔽: 上行流量" ;;
	6) "$TERMINAL" -e "$EDITOR" "$0" ;;
esac

update() {
    sum=0
    for arg; do
        read -r i < "$arg"
        sum=$(( sum + i ))
    done
    cache=${XDG_CACHE_HOME:-$HOME/.cache}/${1##*/}
    [ -f "$cache" ] && read -r old < "$cache" || old=0
    printf %d\\n "$sum" > "$cache"
    printf %d\\n $(( sum - old ))
}

RX=$(update /sys/class/net/[ew]*/statistics/rx_bytes)
TX=$(update /sys/class/net/[ew]*/statistics/tx_bytes)

## 换算单位
#if [[ $RX -lt 1024 ]];then
#    # 如果接收速率小于1024,则单位为B/s
#    RX="${RX}B/s"
#elif [[ $RX -gt 1048576 ]];then
#    # 否则如果接收速率大于 1048576,则改变单位为MB/s
#    RX=$(echo $RX | awk '{print $1/1048576 "MB/s"}')
#else
#    # 否则如果接收速率大于1024但小于1048576,则单位为KB/s
#    RX=$(echo $RX | awk '{print $1/1024 "KB/s"}')
#fi
#
## 换算单位
#if [[ $TX -lt 1024 ]];then
#    # 如果发送速率小于1024,则单位为B/s
#    TX="${TX}B/s"
#elif [[ $TX -gt 1048576 ]];then
#    # 否则如果发送速率大于 1048576,则改变单位为MB/s
#    TX=$(echo $TX | awk '{print $1/1048576 "MB/s"}')
#else
#    # 否则如果发送速率大于1024但小于1048576,则单位为KB/s
#    TX=$(echo $TX | awk '{print $1/1024 "KB/s"}')
#fi
#
#printf "🔼:$TX 🔽:$RX"


# 换算单位
if [[ $RX -lt 1024 ]]; then
    # 如果接收速率小于1024,则显示为1KB/s
    RX="1KB/s"
elif [[ $RX -gt 1048576 ]]; then
    # 否则如果接收速率大于1048576,则改变单位为MB/s，不显示小数点后位数
    RX=$(echo $RX | awk '{printf "%.0f MB/s", $1/1048576}')
else
    # 否则如果接收速率大于1024但小于1048576,则单位为KB/s，不显示小数点后位数
    RX=$(echo $RX | awk '{printf "%.0f KB/s", $1/1024}')
fi

# 换算单位
if [[ $TX -lt 1024 ]]; then
    # 如果发送速率小于1024,则显示为1KB/s
    TX="1KB/s"
elif [[ $TX -gt 1048576 ]]; then
    # 否则如果发送速率大于1048576,则改变单位为MB/s，不显示小数点后位数
    TX=$(echo $TX | awk '{printf "%.0f MB/s", $1/1048576}')
else
    # 否则如果发送速率大于1024但小于1048576,则单位为KB/s，不显示小数点后位数
    TX=$(echo $TX | awk '{printf "%.0f KB/s", $1/1024}')
fi

printf "🔼:%s 🔽:%s\n" "$TX" "$RX"
